#!/bin/bash

# 创建项目根目录
mkdir -p audio_retrieval_system

# 创建项目子目录结构
mkdir -p audio_retrieval_system/.idea
mkdir -p audio_retrieval_system/config
mkdir -p audio_retrieval_system/data/db
mkdir -p audio_retrieval_system/data/music_library
mkdir -p audio_retrieval_system/src/fingerprinting
mkdir -p audio_retrieval_system/src/matching
mkdir -p audio_retrieval_system/scripts

# 创建配置文件
touch audio_retrieval_system/config/config.yaml

# 在音乐库中添加说明文件
echo "请将您的WAV/MP3音乐文件放在此目录中" > audio_retrieval_system/data/music_library/README.txt

# 创建Python包初始化文件
touch audio_retrieval_system/src/__init__.py
touch audio_retrieval_system/src/fingerprinting/__init__.py
touch audio_retrieval_system/src/matching/__init__.py

# 创建源代码文件
touch audio_retrieval_system/src/database.py
touch audio_retrieval_system/src/fingerprinting/processor.py
touch audio_retrieval_system/src/fingerprinting/hashing.py
touch audio_retrieval_system/src/matching/scorer.py

# 创建脚本文件
touch audio_retrieval_system/scripts/01_initialize_db.py
touch audio_retrieval_system/scripts/02_fingerprint_library.py
touch audio_retrieval_system/scripts/03_recognize_sample.py

# 创建依赖文件
touch audio_retrieval_system/requirements.txt

# 输出成功信息
echo "项目结构创建完成！"
echo "项目路径: $(pwd)/audio_retrieval_system"
echo "请将您的音乐文件放入: $(pwd)/audio_retrieval_system/data/music_library"